
import ferramentas.GeradorCSV;
import ferramentas.Query;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.sql.SQLException;
import java.text.ParseException;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import pessoal.Curso;
import pessoal.Inscrito;

/**
 *
 * @author ciar
 */
public class gerarCSV extends HttpServlet {

    /**
     * Processes requests for both HTTP
     * <code>GET</code> and
     * <code>POST</code> methods.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException, SQLException, ParseException {
        response.setContentType("application/download");
        request.setCharacterEncoding("UTF-8");
        response.setCharacterEncoding("UTF-8");
        //PrintWriter out = response.getWriter();

        try {

            String finalidadeCSV = request.getParameter("finalidadeCSV");
            String formato = request.getParameter("formato");

            int numeroCurso = Integer.parseInt(request.getParameter("curso"));

            Query q = new Query();

            Curso curso = q.getCurso(numeroCurso);
            List<Inscrito> inscritosNoCurso = null;
            if (curso.getTipo() == 4) {
                inscritosNoCurso = q.getInscritosCursoOrientador(numeroCurso);

            } else {
                inscritosNoCurso = q.getInscritos(numeroCurso);
            }

            Collections.sort(inscritosNoCurso, new Comparator() {
                @Override
                public int compare(Object o1, Object o2) {
                    Inscrito i1 = (Inscrito) o1;
                    Inscrito i2 = (Inscrito) o2;
                    return i1.getNome().compareToIgnoreCase(i2.getNome());
                }
            });

            String caminhoArquivoCSV = System.getProperty("java.io.tmpdir") + System.getProperty("file.separator");

            GeradorCSV geradorCVS = new GeradorCSV(finalidadeCSV);
            if (finalidadeCSV.equals("sisuab")) {
                int polo = Integer.parseInt(request.getParameter("polo"));
                switch (polo) {
                    case 1:
                        geradorCVS.setPolo("Anápolis");
                        geradorCVS.setNomeArquivo("anapolis.csv");
                        break;
                    case 2:
                        geradorCVS.setPolo("Cezarina");
                        geradorCVS.setNomeArquivo("cezarina.csv");
                        break;
                    case 3:
                        geradorCVS.setPolo("Formosa");
                        geradorCVS.setNomeArquivo("formosa.csv");
                        break;
                    case 4:
                        geradorCVS.setPolo("Uruaçu");
                        geradorCVS.setNomeArquivo("uruacu.csv");
                        break;
                }
            }
            File arquivoCSV = null;
            if (formato.equals("excel")) {
                arquivoCSV = geradorCVS.gerarListaCSVISO(inscritosNoCurso, curso, caminhoArquivoCSV);
            } else {
                arquivoCSV = geradorCVS.gerarListaCSVUTF(inscritosNoCurso, curso, caminhoArquivoCSV);
            }


            int length = (int) arquivoCSV.length();
            ServletOutputStream op = response.getOutputStream();
            response.setContentLength((int) length);
            response.setHeader("Content-Disposition", "attachment; filename=\"" + geradorCVS.getNomeArquivoCSV() + "\"");
            byte[] bbuf = new byte[32767];
            DataInputStream in = new DataInputStream(new FileInputStream(arquivoCSV));

            while ((in != null) && ((length = in.read(bbuf)) != -1)) {
                op.write(bbuf, 0, length);
            }

            in.close();
            op.flush();

        } finally {
            //out.close();
        }
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /**
     * Handles the HTTP
     * <code>GET</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        try {
            processRequest(request, response);
        } catch (SQLException ex) {
            Logger.getLogger(gerarCSV.class.getName()).log(Level.SEVERE, null, ex);
        } catch (ParseException ex) {
            Logger.getLogger(gerarCSV.class.getName()).log(Level.SEVERE, null, ex);
        }
    }

    /**
     * Handles the HTTP
     * <code>POST</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        try {
            processRequest(request, response);
        } catch (SQLException ex) {
            Logger.getLogger(gerarCSV.class.getName()).log(Level.SEVERE, null, ex);
        } catch (ParseException ex) {
            Logger.getLogger(gerarCSV.class.getName()).log(Level.SEVERE, null, ex);
        }
    }

    /**
     * Returns a short description of the servlet.
     *
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>
}
